package com.ruoyi.system.domain.dto;

import com.alibaba.fastjson2.annotation.JSONField;
import lombok.Data;
import java.util.List;

/**
 * @Auther Yulong Liu
 * @Date 2023/12/12 15:02
 * @Description
 **/

@Data
public class MultibaseQueryDto {

    @JSONField(name="session_id")
    String sessionId;

    @JSONField(name="query")
    String query;

    @JSONField(name="knowledge_base_names")
    List<String> knowledge_base_names;

    @JSONField(name="top_k")
    Integer topK;

    @JSONField(name="score_threshold")
    Float scoreThreshold;

    @JSONField(name="history")
    List<MessagesDto> history;

    @JSONField(name="stream")
    Boolean stream;

    @JSONField(name="modal_name")
    String modelName;

    @JSONField(name="temperature")
    Float temperature;

    @JSONField(name="max_tokens")
    Integer maxTokens;

    @JSONField(name="prompt_name")
    String promptName;

    @JSONField(name="prompt")
    String prompt;


    List<Long> kbIds;

}
